package org.csu.onlineedubackend.model.spark;

import lombok.Data;
import org.csu.onlineedubackend.model.AIChatMessage;

import java.util.List;

@Data
public class SparkRequest {
    private Header header;
    private Parameter parameter;
    private Payload payload;
    
    @Data
    public static class Header {
        private String app_id;
        private String uid;
    }
    
    @Data
    public static class Parameter {
        private Chat chat;
        
        @Data
        public static class Chat {
            private String domain;
            private Float temperature;
            private Integer top_k;
            private Integer max_tokens;
        }
    }
    
    @Data
    public static class Payload {
        private Message message;
        
        @Data
        public static class Message {
            private List<AIChatMessage> text;
        }
    }
} 